<template>
  <div class="app-container">
    <el-form :model="queryParams" ref="queryFormRef" :inline="true" v-show="showSearch">
      <el-form-item label="角色名称" prop="roleName">
        <el-input
          v-model="queryParams.roleName"
          placeholder="请输入角色名称"
          clearable
          style="width: 240px"
          @keyup.enter="handleQuery"
        />
      </el-form-item>
      <el-form-item label="权限字符" prop="roleKey">
        <el-input
          v-model="queryParams.roleKey"
          placeholder="请输入权限字符"
          clearable
          style="width: 240px"
          @keyup.enter="handleQuery"
        />
      </el-form-item>
      <el-form-item label="状态" prop="status">
        <el-select
          v-model="queryParams.status"
          placeholder="角色状态"
          clearable
          style="width: 240px"
        >
          <el-option
            v-for="dict in dict?.type?.sys_normal_disable"
            :key="dict.value"
            :label="dict.label"
            :value="dict.value"
          />
        </el-select>
      </el-form-item>
      <el-form-item label="创建时间">
        <el-date-picker
          v-model="dateRange"
          style="width: 240px"
          value-format="yyyy-MM-dd"
          type="daterange"
          range-separator="-"
          start-placeholder="开始日期"
          end-placeholder="结束日期"
        ></el-date-picker>
      </el-form-item>
      <el-form-item>
        <el-button type="primary" icon="Search" @click="handleQuery">搜索</el-button>
        <el-button icon="Refresh" @click="resetQuery">重置</el-button>
      </el-form-item>
    </el-form>

    <el-row :gutter="10" class="mb8">
      <el-col :span="1.5">
        <el-button
          type="primary"
          plain
          icon="Plus"
          @click="handleAdd"
          v-hasPermi="['system:role:add']"
          >新增</el-button
        >
      </el-col>
      <el-col :span="1.5">
        <el-button
          type="success"
          plain
          icon="Edit"
          :disabled="single"
          @click="handleUpdate"
          v-hasPermi="['system:role:edit']"
          >修改</el-button
        >
      </el-col>
      <el-col :span="1.5">
        <el-button
          type="danger"
          plain
          icon="Delete"
          :disabled="multiple"
          @click="handleDelete"
          v-hasPermi="['system:role:remove']"
          >删除</el-button
        >
      </el-col>
      <el-col :span="1.5">
        <el-button
          type="warning"
          plain
          icon="Download"
          @click="handleExport"
          v-hasPermi="['system:role:export']"
          >导出</el-button
        >
      </el-col>
      <right-toolbar v-model:showSearch="showSearch" @queryTable="getList"></right-toolbar>
    </el-row>

    <el-table v-loading="loading" :data="roleList" @selection-change="handleSelectionChange">
      <el-table-column type="selection" width="55" align="center" />
      <el-table-column label="角色编号" prop="roleId" width="120" />
      <el-table-column label="角色名称" prop="roleName" :show-overflow-tooltip="true" width="150" />
      <el-table-column label="权限字符" prop="roleKey" :show-overflow-tooltip="true" width="150" />
      <el-table-column label="显示顺序" prop="roleSort" width="100" />
      <el-table-column label="状态" align="center" width="100">
        <template #default="scope">
          <el-switch
            v-model="scope.row.status"
            active-value="0"
            inactive-value="1"
            @change="handleStatusChange(scope.row)"
          ></el-switch>
        </template>
      </el-table-column>
      <el-table-column label="创建时间" align="center" prop="createTime" width="180">
        <template #default="scope">
          <span>{{ parseTime(scope.row.createTime) }}</span>
        </template>
      </el-table-column>
      <el-table-column label="创建者" align="center" prop="deptName" width="200" />
      <el-table-column label="操作" align="center" class-name="small-padding fixed-width">
        <template #default="scope">
          <div v-if="scope.row.roleId !== 1">
            <el-button
              type="text"
              icon="Edit"
              @click="handleUpdate(scope.row)"
              v-hasPermi="['system:role:edit']"
              >修改</el-button
            >
            <el-button
              type="text"
              icon="Delete"
              @click="handleDelete(scope.row)"
              v-hasPermi="['system:role:remove']"
              >删除</el-button
            >
            <el-dropdown
              @command="(command) => handleCommand(command, scope.row)"
              v-hasPermi="['system:role:edit']"
            >
              <el-button type="text" icon="ArrowRight">更多</el-button>
              <template #default:dropdown>
                <el-dropdown-menu>
                  <el-dropdown-item
                    command="handleDataScope"
                    icon="CircleCheck"
                    v-hasPermi="['system:role:edit']"
                    >数据权限</el-dropdown-item
                  >
                  <el-dropdown-item
                    command="handleAuthUser"
                    icon="User"
                    v-hasPermi="['system:role:edit']"
                    >分配用户</el-dropdown-item
                  >
                </el-dropdown-menu>
              </template>
            </el-dropdown>
          </div>
        </template>
      </el-table-column>
    </el-table>

    <pagination
      v-show="total > 0"
      :total="total"
      v-model:page="queryParams.pageNum"
      v-model:limit="queryParams.pageSize"
      @pagination="getList"
    />

    <!-- 添加或修改角色配置对话框 -->
    <el-dialog :title="title" v-model="open" width="500px" append-to-body>
      <el-form ref="formRef" :model="form" :rules="rules" label-width="100px">
        <el-form-item label="角色名称" prop="roleName">
          <el-input v-model="form.roleName" placeholder="请输入角色名称" />
        </el-form-item>
        <el-form-item v-if="showRoleByte" v-show="false">
          <template #default:label>
            <span>
              <el-tooltip
                content="控制器中定义的权限字符，如：@PreAuthorize(`@ss.hasRole('admin')`)"
                placement="top"
              >
                <i class="el-icon-question"></i>
              </el-tooltip>
              权限字符
            </span>
          </template>
          <el-input v-model="form.roleKey" placeholder="请输入权限字符" />
        </el-form-item>
        <!--        <el-form-item label="角色顺序" prop="roleSort">-->
        <!--          <el-input-number v-model="form.roleSort" controls-position="right" :min="0" />-->
        <!--        </el-form-item>-->
        <el-form-item label="状态">
          <el-radio-group v-model="form.status">
            <el-radio
              v-for="dict in dict?.type?.sys_normal_disable"
              :key="dict.value"
              :label="dict.value"
              >{{ dict.label }}</el-radio
            >
          </el-radio-group>
        </el-form-item>
        <el-form-item label="菜单权限">
          <el-checkbox v-model="menuExpand" @change="handleCheckedTreeExpand($event, 'menu')"
            >展开/折叠</el-checkbox
          >
          <el-checkbox
            v-show="false"
            v-model="menuNodeAll"
            @change="handleCheckedTreeNodeAll($event, 'menu')"
            >全选/全不选</el-checkbox
          >
          <!--          <el-checkbox-->
          <!--            v-model="form.menuCheckStrictly"-->
          <!--            @change="handleCheckedTreeConnect($event, 'menu')"-->
          <!--            >父子联动</el-checkbox-->
          <!--          >-->
          <el-tree
            class="tree-border"
            :data="menuOptions"
            show-checkbox
            ref="menuRef"
            node-key="id"
            :check-strictly="!form.menuCheckStrictly"
            empty-text="加载中，请稍候"
            :props="defaultProps"
          ></el-tree>
        </el-form-item>
        <el-form-item label="备注">
          <el-input v-model="form.remark" type="textarea" placeholder="请输入内容"></el-input>
        </el-form-item>
      </el-form>
      <template #footer>
        <div class="dialog-footer">
          <el-button type="primary" @click="submitForm">确 定</el-button>
          <el-button @click="cancel">取 消</el-button>
        </div>
      </template>
    </el-dialog>

    <!-- 分配角色数据权限对话框 -->
    <el-dialog :title="title" v-model="openDataScope" width="500px" append-to-body>
      <el-form :model="form" label-width="80px">
        <el-form-item label="角色名称">
          <el-input v-model="form.roleName" :disabled="true" />
        </el-form-item>
        <el-form-item label="权限字符">
          <el-input v-model="form.roleKey" :disabled="true" />
        </el-form-item>
        <el-form-item label="权限范围">
          <el-select v-model="form.dataScope" @change="dataScopeSelectChange">
            <el-option
              v-for="item in dataScopeOptions"
              :key="item.value"
              :label="item.label"
              :value="item.value"
            ></el-option>
          </el-select>
        </el-form-item>
        <el-form-item label="数据权限" v-show="form.dataScope == 2">
          <el-checkbox v-model="deptExpand" @change="handleCheckedTreeExpand($event, 'dept')"
            >展开/折叠</el-checkbox
          >
          <el-checkbox v-model="deptNodeAll" @change="handleCheckedTreeNodeAll($event, 'dept')"
            >全选/全不选</el-checkbox
          >
          <!--          <el-checkbox-->
          <!--            v-model="form.deptCheckStrictly"-->
          <!--            @change="handleCheckedTreeConnect($event, 'dept')"-->
          <!--            >父子联动</el-checkbox-->
          <!--          >-->
          <el-tree
            class="tree-border"
            :data="deptOptions"
            show-checkbox
            default-expand-all
            ref="deptRef"
            node-key="id"
            :check-strictly="!form.deptCheckStrictly"
            empty-text="加载中，请稍候"
            :props="defaultProps"
          ></el-tree>
        </el-form-item>
      </el-form>
      <template #footer>
        <div class="dialog-footer">
          <el-button type="primary" @click="submitDataScope">确 定</el-button>
          <el-button @click="cancelDataScope">取 消</el-button>
        </div>
      </template>
    </el-dialog>
  </div>
</template>

<script setup lang="ts">
import { ref, reactive, getCurrentInstance, onMounted } from 'vue'
import {
  listRole,
  getRole,
  delRole,
  addRole,
  updateRole,
  dataScope,
  changeRoleStatus,
  deptTreeSelect,
} from '@/api/system/role'
import { treeselect as menuTreeselect, roleMenuTreeselect } from '@/api/system/menu'
import { ElMessage, ElMessageBox, type FormInstance } from 'element-plus'
import { parseTime, addDateRange } from '@/utils/cj'
import { useDict } from '@/utils/dict'
import { useStore } from 'vuex'

const { proxy } = getCurrentInstance() as any

const { dict }: any = useDict({
  type: 'sys_normal_disable',
})

const store = useStore()

// 遮罩层
const loading = ref(true)
// 选中数组
const ids = ref<Array<number>>([])
// 非单个禁用
const single = ref(true)
// 非多个禁用
const multiple = ref(true)
// 显示搜索条件
const showSearch = ref(true)
// 总条数
const total = ref(0)
// 角色表格数据
const roleList = ref<Array<any>>([])
// 弹出层标题
const title = ref('')
// 是否显示弹出层
const open = ref(false)
// 是否显示弹出层（数据权限）
const openDataScope = ref(false)
const menuExpand = ref(false)
const menuNodeAll = ref(false)
const deptExpand = ref(true)
const deptNodeAll = ref(false)
const showRoleByte = ref(true)
// 日期范围
const dateRange = ref<Array<string>>([])
// 数据范围选项
const dataScopeOptions = ref([
  {
    value: '1',
    label: '全部数据权限',
  },
  {
    value: '2',
    label: '自定数据权限',
  },
  {
    value: '3',
    label: '本公司数据权限',
  },
  {
    value: '4',
    label: '本公司及以下数据权限',
  },
  {
    value: '5',
    label: '仅本人数据权限',
  },
])
// 菜单列表
const menuOptions = ref<Array<any>>([])
// 部门列表
const deptOptions = ref<Array<any>>([])
// 查询参数
const queryParams = reactive({
  pageNum: 1,
  pageSize: 10,
  roleName: undefined,
  roleKey: undefined,
  status: undefined,
})
// 表单参数
const form = reactive<Record<string, any>>({
  roleId: undefined,
  roleName: undefined,
  roleKey: undefined,
  roleSort: 9999,
  status: '0',
  menuIds: [],
  deptIds: [],
  menuCheckStrictly: true,
  deptCheckStrictly: true,
  remark: undefined,
  dataScope: undefined,
  userId: undefined,
})
const defaultProps = {
  children: 'children',
  label: 'label',
}
// 表单校验
const rules = {
  roleName: [{ required: true, message: '角色名称不能为空', trigger: 'blur' }],
  roleKey: [{ required: true, message: '权限字符不能为空', trigger: 'blur' }],
  // roleSort: [
  //   { required: true, message: "角色顺序不能为空", trigger: "blur" }
  // ]
}

// Refs
const queryFormRef = ref<FormInstance>()
const formRef = ref<FormInstance>()
const menuRef = ref()
const deptRef = ref()

/** 查询角色列表 */
const getList = () => {
  loading.value = true
  listRole(addDateRange(queryParams, dateRange.value)).then((response: any) => {
    roleList.value = response.rows
    total.value = response.total
    loading.value = false
  })
}

/** 查询菜单树结构 */
const getMenuTreeselect = () => {
  menuTreeselect().then((response: any) => {
    menuOptions.value = response.data
  })
}

// 所有菜单节点数据
const getMenuAllCheckedKeys = () => {
  // 目前被选中的菜单节点
  const checkedKeys = menuRef.value?.getCheckedKeys()
  // 半选中的菜单节点
  const halfCheckedKeys = menuRef.value?.getHalfCheckedKeys()
  if (checkedKeys && halfCheckedKeys) {
    checkedKeys.unshift(...halfCheckedKeys)
  }
  return checkedKeys
}

// 所有部门节点数据
const getDeptAllCheckedKeys = () => {
  // 目前被选中的部门节点
  const checkedKeys = deptRef.value?.getCheckedKeys()
  // 半选中的部门节点
  const halfCheckedKeys = deptRef.value?.getHalfCheckedKeys()
  if (checkedKeys && halfCheckedKeys) {
    checkedKeys.unshift(...halfCheckedKeys)
  }
  return checkedKeys
}

/** 根据角色ID查询菜单树结构 */
const getRoleMenuTreeselect = (roleId: string) => {
  return roleMenuTreeselect(roleId).then((response: any) => {
    menuOptions.value = response.menus
    return response
  })
}

/** 根据角色ID查询部门树结构 */
const getDeptTree = (roleId: string) => {
  return deptTreeSelect(roleId).then((response: any) => {
    deptOptions.value = response.depts
    return response
  })
}

// 角色状态修改
const handleStatusChange = (row: any) => {
  const text = row.status === '0' ? '启用' : '停用'
  ElMessageBox.confirm('确认要"' + text + '""' + row.roleName + '"角色吗？', '警告', {
    confirmButtonText: '确定',
    cancelButtonText: '取消',
    type: 'warning',
  })
    .then(() => {
      return changeRoleStatus(row.roleId, row.status)
    })
    .then(() => {
      ElMessage.success(text + '成功')
    })
    .catch(() => {
      row.status = row.status === '0' ? '1' : '0'
    })
}

// 取消按钮
const cancel = () => {
  open.value = false
  reset()
}

// 取消按钮（数据权限）
const cancelDataScope = () => {
  openDataScope.value = false
  reset()
}

// 表单重置
const reset = () => {
  if (menuRef.value) {
    menuRef.value.setCheckedKeys([])
  }
  menuExpand.value = false
  menuNodeAll.value = false
  deptExpand.value = true
  deptNodeAll.value = false
  form.roleId = undefined
  form.roleName = undefined
  form.roleKey = undefined
  form.roleSort = 9999
  form.status = '0'
  form.menuIds = []
  form.deptIds = []
  form.menuCheckStrictly = true
  form.deptCheckStrictly = true
  form.remark = undefined
  form.dataScope = undefined
  form.userId = undefined
  formRef.value?.resetFields()
}

/** 搜索按钮操作 */
const handleQuery = () => {
  queryParams.pageNum = 1
  getList()
}

/** 重置按钮操作 */
const resetQuery = () => {
  dateRange.value = []
  queryFormRef.value?.resetFields()
  handleQuery()
}

// 多选框选中数据
const handleSelectionChange = (selection: any) => {
  ids.value = selection.map((item: any) => item.roleId)
  single.value = selection.length != 1
  multiple.value = !selection.length
}

// 更多操作触发
const handleCommand = (command: string, row: any) => {
  switch (command) {
    case 'handleDataScope':
      handleDataScope(row)
      break
    case 'handleAuthUser':
      handleAuthUser(row)
      break
    default:
      break
  }
}

// 树权限（展开/折叠）
const handleCheckedTreeExpand = (value: any, type: string) => {
  if (type == 'menu') {
    const treeList = menuOptions.value
    for (let i = 0; i < treeList.length; i++) {
      menuRef.value.store.nodesMap[treeList[i].id].expanded = value
    }
  } else if (type == 'dept') {
    const treeList = deptOptions.value
    for (let i = 0; i < treeList.length; i++) {
      deptRef.value.store.nodesMap[treeList[i].id].expanded = value
    }
  }
}

// 树权限（全选/全不选）
const handleCheckedTreeNodeAll = (value: any, type: string) => {
  if (type == 'menu') {
    menuRef.value?.setCheckedNodes(value ? menuOptions.value : [])
  } else if (type == 'dept') {
    deptRef.value?.setCheckedNodes(value ? deptOptions.value : [])
  }
}

// 树权限（父子联动）
const handleCheckedTreeConnect = (value: any, type: string) => {
  if (type == 'menu') {
    form.menuCheckStrictly = value ? true : false
  } else if (type == 'dept') {
    form.deptCheckStrictly = value ? true : false
  }
}

/** 新增按钮操作 */
const handleAdd = () => {
  if (store?.state?.roles?.includes('merchants_admin')) {
    showRoleByte.value = false
  }
  reset()
  getMenuTreeselect()
  open.value = true
  title.value = '添加角色'
}

/** 修改按钮操作 */
const handleUpdate = (row: any) => {
  if (store?.state?.roles?.includes('merchants_admin')) {
    showRoleByte.value = false
  }
  reset()
  const roleId = row.roleId || ids.value
  const roleMenu = getRoleMenuTreeselect(roleId)
  getRole(roleId).then((response: any) => {
    Object.assign(form, response.data)
    open.value = true
    proxy.$nextTick(() => {
      roleMenu.then((res: any) => {
        const checkedKeys = res.checkedKeys
        checkedKeys.forEach((v: any) => {
          proxy.$nextTick(() => {
            menuRef.value?.setChecked(v, true, false)
          })
        })
      })
    })
  })
  title.value = '修改角色'
}

/** 选择角色权限范围触发 */
const dataScopeSelectChange = (value: any) => {
  if (value !== '2') {
    deptRef.value?.setCheckedKeys([])
  }
}

/** 分配数据权限操作 */
const handleDataScope = (row: any) => {
  reset()
  const deptTree = getDeptTree(row.roleId)
  getRole(row.roleId).then((response: any) => {
    Object.assign(form, response.data)
    openDataScope.value = true
    proxy.$nextTick(() => {
      deptTree.then((res: any) => {
        deptRef.value?.setCheckedKeys(res.checkedKeys)
      })
    })
  })
  title.value = '分配数据权限'
}

/** 分配用户操作 */
const handleAuthUser = (row: any) => {
  const roleId = row.roleId
  proxy.$router.push('/system/role-auth/user/' + roleId)
}

/** 提交按钮 */
const submitForm = () => {
  formRef.value?.validate((valid: any) => {
    if (valid) {
      if (form.roleId != undefined) {
        form.menuIds = getMenuAllCheckedKeys()
        updateRole(form).then((response: any) => {
          ElMessage.success('修改成功')
          open.value = false
          getList()
        })
      } else {
        form.menuIds = getMenuAllCheckedKeys()
        form.userId = store.state.user.id
        addRole(form).then((response: any) => {
          ElMessage.success('新增成功')
          open.value = false
          getList()
        })
      }
    }
  })
}

/** 提交按钮（数据权限） */
const submitDataScope = () => {
  if (form.roleId != undefined) {
    form.deptIds = getDeptAllCheckedKeys()
    dataScope(form).then((response: any) => {
      ElMessage.success('修改成功')
      openDataScope.value = false
      getList()
    })
  }
}

/** 删除按钮操作 */
const handleDelete = (row: any) => {
  const roleIds = row.roleId || ids.value
  ElMessageBox.confirm('是否确认删除角色编号为"' + roleIds + '"的数据项？')
    .then(() => {
      return delRole(roleIds)
    })
    .then(() => {
      getList()
      ElMessage.success('删除成功')
    })
    .catch(() => {})
}

/** 导出按钮操作 */
const handleExport = () => {
  proxy.download(
    'system/role/export',
    {
      ...queryParams,
    },
    `role_${new Date().getTime()}.xlsx`,
  )
}

onMounted(() => {
  getList()
})
</script>
